import pandas as pd

import os

'''
@param sale_data_file 销售数据文件
@param weather_data_file 天气数据文件
@dst_dir 合并后文件存放路径，文件名和销售数据文件名一致
'''


def build(sale_data_file, weather_data_file, dst_dir):
    if not os.path.exists(sale_data_file):
        print('sale data file %s not exist.' % sale_data_file)
        return
    if not os.path.exists(weather_data_file):
        print('weather data file %s not exist.' % weather_data_file)
        return
    if not os.path.exists(dst_dir):
        os.makedirs(dst_dir)

    file_name = os.path.basename(sale_data_file)
    sale_data = pd.read_csv(sale_data_file, dtype={'商品编码': str})
    weather_data = pd.read_csv(weather_data_file)
    merged_data = pd.merge(sale_data, weather_data, on='日期')
    merged_data.to_csv(dst_dir + '/' + file_name, index=False)


if __name__ == "__main__":
    build('./sale/001488.csv', './weather.csv', './data')
    #6927128225877.csv
